#coding=utf-8
#unittest框架编写单元测试用例
from calculator import Count
import unittest

class TestAdd(unittest.TestCase):  #继承unittest中TestCase类
    #测试用例执行前的初始化工作
    def setUp(self):
        print "test start"
    def test_add(self):
        a=Count(1,2)
        #TestCase 中带assertEqual()方法
        self.assertEqual(a.add(),3)
    #测试用例执行执行完,进行提示
    def test_add2(self):
        a=Count(3,2)
        #TestCase 中带assertEqual()方法
        self.assertEqual(a.add(),5)
    def tearDown(self):
        print "test end"
#__name__模块的内置属性   .py文件可以作为模块被调用,也可直接使用
#如果等于main表示直接使用
if __name__ == "__main__":
    unittest.main()
